
package br.com.mackinvest.dao;

import static java.lang.System.out;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Timestamp;
import java.util.Set;
import java.util.TreeMap;
import java.util.TreeSet;

import br.com.mackinvest.constants.Constants;
import br.com.mackinvest.model.Cliente;
import br.com.mackinvest.model.Investimento;
import br.com.mackinvest.model.SaldosInvestimento;

public class SaldosInvestimentoMySQLDAO implements GenericDAO<SaldosInvestimento> {
	private PreparedStatement statement;
	private String query;
	private Connection con;
	
	public SaldosInvestimentoMySQLDAO() {
		con = DBUtil.getConnection();
	}
	
	public void salva(SaldosInvestimento model) 
	{
		query = "INSERT INTO tbsaldosinvestimento(saldo, dataUltimaTransacao, id_cliente, id_investimento) VALUES (?,?,?,?)";
		try{
			statement = con.prepareStatement(query);
			statement.setFloat(1, model.getSaldo());
			statement.setTimestamp(2, new Timestamp( model.getDataUltimaTransacao().getTime()));
			
			if ( model.getCliente() != null ) 
				statement.setInt(3, model.getCliente().getId() );
			else
				statement.setInt(3, 0 );
			
			if ( model.getInvestimento() != null )
				statement.setInt(4, model.getInvestimento().getId() );
			else
				statement.setInt(4, 0 );
		
			statement.executeUpdate();
			
		}
		catch(SQLException ex)
		{
			out.println(Constants.ERRO_INSERT_NAO_REALIZADO);
			out.println(ex);
		}	 
	}
	 
	public void deleta(SaldosInvestimento model) {
		try
		{
			query = "DELETE FROM tbsaldosinvestimento WHERE id=?";			
			statement = con.prepareStatement(query);
			statement.setInt(1, model.getId());
			statement.executeUpdate();
		}
		catch(SQLException ex)
		{
			out.println(Constants.ERRO_DELETE_NAO_REALIZADO);
			out.println(ex);
		}
	}
	 
	public void atualiza(SaldosInvestimento model) 
	{
		query = "UPDATE tbsaldosinvestimento SET saldo = ?, dataUltimaTransacao = ?, id_cliente = ?, id_investimento = ? WHERE id=?  ";
		try{
			statement = con.prepareStatement(query);
			statement.setFloat(1, model.getSaldo());
			statement.setTimestamp(2, new Timestamp( model.getDataUltimaTransacao().getTime()));
			
			if ( model.getCliente() != null ) 
				statement.setInt(3, model.getCliente().getId() );
			else
				statement.setInt(3, 0 );
			
			if ( model.getInvestimento() != null )
				statement.setInt(4, model.getInvestimento().getId() );
			else
				statement.setInt(4, 0 );
		
			statement.executeUpdate();
			
		}
		catch(SQLException ex)
		{
			out.println(Constants.ERRO_INSERT_NAO_REALIZADO);
			out.println(ex);
		}	 
	}
 
 
	public TreeMap<Integer, SaldosInvestimento> retornaTodos() 
	{
		query = "SELECT * FROM tbSaldosInvestimentos ";
		TreeMap<Integer, SaldosInvestimento> saldos = new TreeMap<Integer, SaldosInvestimento>();
		SaldosInvestimento saldo = null; 
		try
		{
			statement = con.prepareStatement(query);
			
			ResultSet rs = statement.executeQuery();
			while(rs.next()){
				
				saldo = new SaldosInvestimento(rs.getInt("id"),rs.getFloat("saldo"),rs.getDate("dataUltimaTransacao"));
			
				// buscando o investimento que est� relacionado com esse saldo
				Investimento investimentoExemplo = new Investimento();
				investimentoExemplo.setId( rs.getInt("id_investimento") );
				
				saldo.setInvestimento( (new InvestimentoMySQLDAO()).procuraPorExemplo( investimentoExemplo ) );

				// buscando o cliente que est� relacionado com esse saldo...
				Cliente clienteExexmplo = new Cliente();
				clienteExexmplo.setId( rs.getInt("id_cliente") );
				
				saldo.setCliente( (new ClienteMySQLDAO()).procuraPorExemplo( clienteExexmplo ) );
				saldo.setPersistido(true);
				
				saldos.put(saldo.getId(),saldo);
			}
		}
		catch(SQLException ex)
		{
			out.println(Constants.ERRO_REGISTRO_NAO_ENCONTRADO);
			out.println(ex);			
		}
		return saldos;
	}
	 
	public SaldosInvestimento procuraPorExemplo(SaldosInvestimento model) {
		return null;
	}

	public Set<SaldosInvestimento> procuraPorCliente(Cliente cliente) 
	{
		query = "SELECT * FROM tbSaldosInvestimentos WHERE id_cliente=?";
		Set<SaldosInvestimento> saldos = new TreeSet<SaldosInvestimento>();
		SaldosInvestimento saldo = null; 
		try
		{
			statement = con.prepareStatement(query);
			statement.setInt(1, cliente.getId());
			
			ResultSet rs = statement.executeQuery();
			while(rs.next()){
				
				saldo = new SaldosInvestimento(rs.getInt("id"),rs.getFloat("saldo"),rs.getDate("dataUltimaTransacao"));
			
				// buscando o investimento que est� relacionado com esse saldo
				Investimento investimentoExemplo = new Investimento();
				investimentoExemplo.setId( rs.getInt("id_investimento") );
				
				saldo.setInvestimento( (new InvestimentoMySQLDAO()).procuraPorExemplo( investimentoExemplo ) );

				// buscando o cliente que esta relacionado com esse saldo...
				Cliente clienteExexmplo = new Cliente();
				clienteExexmplo.setId( rs.getInt("id_cliente") );
				
				saldo.setCliente( (new ClienteMySQLDAO()).procuraPorExemplo( clienteExexmplo ) );
				saldo.setPersistido(true);
				
				saldos.add( saldo );
			}
		}
		catch(SQLException ex)
		{
			out.println(Constants.ERRO_REGISTRO_NAO_ENCONTRADO);
			out.println(ex);			
		}
		return saldos;
	}
}
 
